package main


import "fmt"
import "sync"
import "math/rand"
import "time"

var cond sync.Cond             // 创建全局条件变量

// 生产者
func producer(out chan<- int, idx int) {
	for {
		cond.L.Lock()           	// 条件变量对应互斥锁加锁
		for len(out) == 3 {          	// 产品区满 等待消费者消费
			cond.Wait()             	// 挂起当前协程， 等待条件变量满足，被消费者唤醒
		}
		num := rand.Intn(1000) 	// 产生一个随机数
		out <- num             	// 写入到 channel 中 （生产）
		fmt.Printf("%dth 生产者，产生数据 %3d, 公共区剩余%d个数据\n", idx, num, len(out))
		cond.L.Unlock()             	// 生产结束，解锁互斥锁
		cond.Signal()           	// 唤醒 阻塞的 消费者
		time.Sleep(time.Second)       // 生产完休息一会，给其他协程执行机会
	}
}
//消费者
func consumer1(in <-chan int, idx int) {
	for {
		cond.L.Lock()           	// 条件变量对应互斥锁加锁（与生产者是同一个）
		for len(in) == 0 {      	// 产品区为空 等待生产者生产
			cond.Wait()             	// 挂起当前协程， 等待条件变量满足，被生产者唤醒
			//当前线程睡眠之后，下一条线程被唤醒的有可能也是当前线程，所以不能用if 来判断，应该使用for
		}
		num := <-in                	// 将 channel 中的数据读走 （消费）
		fmt.Printf("---- %dth 消费者, 消费数据 %3d,公共区剩余%d个数据\n", idx, num, len(in))
		cond.L.Unlock()             	// 消费结束，解锁互斥锁
		cond.Signal()           	// 唤醒 阻塞的 生产者
		time.Sleep(time.Millisecond * 500)    	//消费完 休息一会，给其他协程执行机会
	}
}
func main() {
	rand.Seed(time.Now().UnixNano())  // 设置随机数种子
	quit := make(chan bool)           // 创建用于结束通信的 channel

	product := make(chan int, 3)      // 产品区（公共区）使用channel 模拟
	cond.L = new(sync.Mutex)          // 创建互斥锁和条件变量

	for i := 0; i < 5; i++ {          // 5个消费者
		go producer(product, i+1)
	}
	for i := 0; i < 3; i++ {          // 3个生产者
		go consumer1(product, i+1)
	}
	<-quit                         	// 主协程阻塞 不结束
}
